import tensorflow as tf
from keras.layers import Dense
import os

if __name__ == '__main__':
    os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

    hidden1 = Dense(units=256, activation='relu')
    hidden2 = Dense(units=128, activation='relu')
    hidden3 = Dense(units=64, activation='relu')
    hidden4 = Dense(units=10, activation='relu')

    n = 6
    input = tf.random.normal(shape=[n, 28 * 28])

    hidden1_output = hidden1(input)
    hidden2_output = hidden2(hidden1_output)
    hidden3_output = hidden3(hidden2_output)
    hidden4_output = hidden4(hidden3_output)

    print(hidden4_output)